import unittest

import config
from service.bert_service import sentence_embedding


class BertAsServiceTestt(unittest.TestCase):
    def test_something(self):
        vv = sentence_embedding(["晓看红湿处", "花重锦官城"])
        assert len(vv) == 2
        assert len(vv[0]) == config.BERT_DIMENSION


if __name__ == '__main__':
    unittest.main()
